# 黑马app的基类

from selenium.webdriver.support.wait import WebDriverWait
from utils import UtilsDriver


# 黑马app对象库基类
class BasePage:
    # 获取黑马app的驱动
    def __init__(self):
        self.app_driver = UtilsDriver.get_app_driver()

    # 定义获取元素定位的方法
    def get_element(self, location):
        wait = WebDriverWait(self.app_driver, 10, 1)
        element = wait.until(lambda x: x.find_element(*location))
        return element


# 黑马app操作库基类
class BaseHandle:
    def input_text(self, element, text):
        element.clear()
        element.send_keys(text)

